how to plot a function in python numpy